poj 3278 BFS

  poj3278

  都说水题，都说是自己第一道BFS题，哈哈。通过这题，又对BFS的基本框架有了进一步的认识。

 分三个方向来BFS，主要时还是判断边界问题。

#include <iostream>
#include <fstream>
#include <cstdio>
#include <queue>
#include <memory.h>

#define MAX 100005
using namespace std;

queue<int> q;
int step[MAX]; //跟随数组，记录步数
bool visit[MAX];
int n,k;
int next,head;

int bfs()
{
    q.push(n);
    step[n]=0;
    visit[n]=1;
    while(!q.empty())
    {
        head=q.front();
        q.pop();
        //分三个方向BFS
        for(int i=0; i<3 ;i++)
        {
            if(i==0) next=head-1;
            else if(i==1) next=head+1;
            else  next=head*2;
            if(next>MAX || next<0 )  continue;  //必须先判断，不然会RE，数组越界了
            if(!visit[next])
            {
              q.push(next);
              step[next]=step[head]+1;
              visit[next]=1;
            }
            if(next==k)  return step[next];
        }
    }
    return -1;
}

int main()
{
    memset(visit,0,sizeof(visit));
    scanf("%ld%ld",&n,&k);
    if(n>=k)
    {
        printf("%d",n-k);
    }
    else
    {
       printf("%d",bfs());
    }
    return 0;
}